import matplotlib.pyplot as plt
import numpy as np
from test_4_00 import create_polar_system

# 创建极坐标系（支持负半径）
fig, ax = create_polar_system(r_range=(-3, 3))

# 设置四叶玫瑰线参数
a = 3  # 控制曲线大小

# 生成角度值（密集采样确保曲线平滑）
theta = np.linspace(0, 2*np.pi, 1000)

# 计算四叶玫瑰线半径（两种常见形式）
# 形式1: r = a * sin(2θ)
r1 = a * np.sin(2*theta)

# 形式2: r = a * cos(2θ)
r2 = a * np.cos(2*theta)

# 绘制两种形式的四叶玫瑰线
ax.plot(theta, r1, 'b-', linewidth=2, label='r = a·sin(2θ)')
ax.plot(theta, r2, 'r-', linewidth=2, label='r = a·cos(2θ)')

# 添加图例
ax.legend(loc='upper right', bbox_to_anchor=(1.15, 1.1))

# 添加标题
ax.set_title('四叶玫瑰线 (Four-Leaf Rose)', fontsize=16, pad=20)

# 添加参数说明
ax.text(0.1, 2.5, f'a = {a}', fontsize=12, color='black')

plt.tight_layout()
plt.show()